#ifndef __EPOLL_SERVER_HPP
#define __EPOLL_SERVER_HPP

#include <iostream>
#include <string>
#include <cassert>
#include <functional>
#include "Log.hpp"
#include "Sock.hpp"
#include "Epoll.hpp"

namespace ns_epoll
{
    static const int default_port = 8080;
    static const int gnum = 64;

    class epollServer
    {
        using func_t = std::function<void(std::string)>;
    public:
        epollServer(func_t HandlerRequest, const int &port = default_port)
            : _port(port), _revs_num(gnum), _HandlerRequest(HandlerRequest)
        {
            // 申请空间
            _revs = new struct epoll_event[_revs_num];

            // 创建套接字
            _listensock = Sock::Socket();
            Sock::Bind(_listensock, _port);
            Sock::Listen(_listensock);

            // 创建epoll模型
            _epfd = Epoll::CreateEpoll();
            logMessage(DEBUG, "init success, listensock:%d epfd:%d", _listensock, _epfd);

            // 将lisensock添加到epoll中，让epoll帮我们管理起来
            if (!Epoll::CtlEpoll(_epfd, EPOLL_CTL_ADD, _listensock, EPOLLIN))
                exit(6);
            logMessage(DEBUG, "add listensock to epoll success");
        }

        void Accepter(int listensock)
        {
            std::string clientip;
            uint16_t clientport;
            int sock=Sock::Accept(listensock, &clientip, &clientport);
            if(sock <0 )
            {
                logMessage(WARNING, "accept error!");
                return ;
            }
            // 不能直接读取，不能清楚的知道底层有没有数据
            if (!Epoll::CtlEpoll(_epfd, EPOLL_CTL_ADD, sock, EPOLLIN)) return;
            logMessage(DEBUG, "add new aock : %d to epoll success ", sock);
        }

        void Recver(int sock)
        {
            // 读数据
            char buffer[10240];
            ssize_t n=recv(sock, buffer, sizeof(buffer)-1, 0);
            if(n>0)
            {
                // 假设这里就是读到了一个完整报文
                buffer[n]=0;
                _HandlerRequest(buffer);  // 处理数据
            }
            else if(n==0)
            {
                // 先在epoll中去掉sock
                bool res=Epoll::CtlEpoll(_epfd, EPOLL_CTL_DEL, sock, 0);
                assert(res);
                (void)res;
                // 在close文件
                close(sock);
                logMessage(NORMAL, "client %d quit,me to...", sock);
            }
            else
            {
                // 先在epoll中去掉sock
                bool res=Epoll::CtlEpoll(_epfd, EPOLL_CTL_DEL, sock, 0);
                assert(res);
                // 在close文件
                close(sock);
                logMessage(NORMAL, "client recv %d error, close error sock", sock);                
            }
        }

        void HandlerEvents(int n)
        {
            assert(n>0);
            for(int i=0;i<n; i++)
            {
                uint32_t revents=_revs[i].events;
                int sock = _revs[i].data.fd;
                //读时间就绪
                if(revents & EPOLLIN)
                {
                    if(sock == _listensock) Accepter(_listensock);  //listensock就绪
                    else Recver(sock); // 一般sock 就绪 - read 
                }
            }
        }

        void loopOnce(int timeout)
        {
            int n = Epoll::WaitEpoll(_epfd, _revs, _revs_num, timeout);
            switch (n)
            {
            case 0:
                logMessage(DEBUG, "timeout...");
                break;
            case -1:
                logMessage(WARNING, "epoll wait error:%s", strerror(errno));
                break;
            default:
                // 成功
                logMessage(DEBUG, "Get a event");
                HandlerEvents(n);
                break;
            }
        }

        void start()
        {
            int timeout=1000;
            while (true)
            {
                loopOnce(timeout);
            }
        }

        ~epollServer()
        {
            if (_listensock >= 0)
                close(_listensock);
            if (_epfd)
                close(_epfd);
            if (_revs)
                delete[] _revs;
        }

    private:
        int _listensock;
        int _epfd;
        uint16_t _port;
        struct epoll_event *_revs;
        int _revs_num;
        func_t _HandlerRequest;
    };
}

#endif